Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 23:12:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

use float32 reference for textual inversion (fixes tests on CPU)

Browse source
This commit is contained in:
Pierre Chapuis 2023-09-11 14:13:40 +02:00
parent e5425e2968
commit dd0cca5855
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/foundationals/clip/test_concepts.py
View file

@ -45,7 +45,7 @@ def our_encoder_with_new_concepts(
def ref_sd15_with_new_concepts( def ref_sd15_with_new_concepts(
runwayml_weights_path: Path, test_textual_inversion_path: Path, test_device: torch.device runwayml_weights_path: Path, test_textual_inversion_path: Path, test_device: torch.device
): ):
pipe = StableDiffusionPipeline.from_pretrained(runwayml_weights_path, torch_dtype=torch.float16).to(test_device) # type: ignore pipe = StableDiffusionPipeline.from_pretrained(runwayml_weights_path).to(test_device) # type: ignore
pipe.load_textual_inversion(test_textual_inversion_path / "cat-toy") # type: ignore pipe.load_textual_inversion(test_textual_inversion_path / "cat-toy") # type: ignore
pipe.load_textual_inversion(test_textual_inversion_path / "gta5-artwork") # type: ignore pipe.load_textual_inversion(test_textual_inversion_path / "gta5-artwork") # type: ignore
return pipe return pipe