Spaces:
Running
on
Zero
Running
on
Zero
Update inf.py
Browse files
inf.py
CHANGED
@@ -25,7 +25,7 @@ class InferencePipeline:
|
|
25 |
else:
|
26 |
self.pipe = StableDiffusionXLPipeline.from_pretrained(
|
27 |
self.base_model_id,
|
28 |
-
torch_dtype=torch.float16,
|
29 |
use_auth_token=self.hf_token)
|
30 |
self.pipe = self.pipe.to(self.device)
|
31 |
self.content_lora_model_id = None
|
|
|
25 |
else:
|
26 |
self.pipe = StableDiffusionXLPipeline.from_pretrained(
|
27 |
self.base_model_id,
|
28 |
+
# torch_dtype=torch.float16,
|
29 |
use_auth_token=self.hf_token)
|
30 |
self.pipe = self.pipe.to(self.device)
|
31 |
self.content_lora_model_id = None
|