hysts HF staff commited on
Commit
2df1835
·
1 Parent(s): 88b57b3
Files changed (2) hide show
  1. README.md +1 -1
  2. model.py +2 -6
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🏢
4
  colorFrom: blue
5
  colorTo: pink
6
  sdk: gradio
7
- sdk_version: 3.35.2
8
  app_file: app.py
9
  pinned: false
10
  suggested_hardware: t4-small
 
4
  colorFrom: blue
5
  colorTo: pink
6
  sdk: gradio
7
+ sdk_version: 3.36.1
8
  app_file: app.py
9
  pinned: false
10
  suggested_hardware: t4-small
model.py CHANGED
@@ -19,8 +19,6 @@ sys.path.insert(0, 'multires_textual_inversion')
19
 
20
  from pipeline import MultiResPipeline, load_learned_concepts
21
 
22
- HF_TOKEN = os.getenv('HF_TOKEN')
23
-
24
 
25
  class Model:
26
  def __init__(self):
@@ -28,13 +26,11 @@ class Model:
28
  'cuda:0' if torch.cuda.is_available() else 'cpu')
29
  model_id = 'runwayml/stable-diffusion-v1-5'
30
  if self.device.type == 'cpu':
31
- pipe = MultiResPipeline.from_pretrained(model_id,
32
- use_auth_token=HF_TOKEN)
33
  else:
34
  pipe = MultiResPipeline.from_pretrained(model_id,
35
  torch_dtype=torch.float16,
36
- revision='fp16',
37
- use_auth_token=HF_TOKEN)
38
  self.pipe = pipe.to(self.device)
39
  self.pipe.scheduler = DPMSolverMultistepScheduler(
40
  beta_start=0.00085,
 
19
 
20
  from pipeline import MultiResPipeline, load_learned_concepts
21
 
 
 
22
 
23
  class Model:
24
  def __init__(self):
 
26
  'cuda:0' if torch.cuda.is_available() else 'cpu')
27
  model_id = 'runwayml/stable-diffusion-v1-5'
28
  if self.device.type == 'cpu':
29
+ pipe = MultiResPipeline.from_pretrained(model_id)
 
30
  else:
31
  pipe = MultiResPipeline.from_pretrained(model_id,
32
  torch_dtype=torch.float16,
33
+ revision='fp16')
 
34
  self.pipe = pipe.to(self.device)
35
  self.pipe.scheduler = DPMSolverMultistepScheduler(
36
  beta_start=0.00085,